package com.ywl.leetcode.twoadd;

/**
 * 2.两数相加
 * 给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。
 * 请你将两个数相加，并以相同形式返回一个表示和的链表。
 * 你可以假设除了数字 0 之外，这两个数都不会以 0 开头。
 *
 * 示例1：
 * 2 ——> 4 ——> 3
 * 5 ——> 6 ——> 4
 * ————————————————
 * 7 ——> 0 ——> 8
 * 输入：l1=[2,4,3], l2=[5,6,4]
 * 输出：[7,0,8]
 * 解释：342 + 465 = 807
 *
 * 示例2：
 * 输入：l1=[0], l2=[0]
 * 输出：[0]
 *
 * 示例3：
 * 输入：l1=[2,4,3], l2=[5,6]
 * 输出：[7,0,4]
 * 解释：342 + 65 = 407
 */
public class TwoAdd {

    public static void main(String[] args) {
        // 链表1
        ListNode n1 = new ListNode(2);
        n1.addFirst(4).addFirst(3);
        // 链表2
        ListNode n2 = new ListNode(5);
        n2.addFirst(6);
        // 操作
        TwoAdd two = new TwoAdd();
        ListNode r = two.twoAdd(n1, n2);
        r.display();
    }

    /**
     * 链表实现
     * @param n1
     * @param n2
     * @return
     */
    public ListNode twoAdd(ListNode n1, ListNode n2) {
        ListNode head = new ListNode(0);
        ListNode cur = head;
        int addV = 0;
        while (n1 != null || n2 != null) {
            int v1 = n1 != null ? n1.val : 0;
            int v2 = n2 != null ? n2.val : 0;
            int s = v1 + v2 + addV;
            addV = s / 10;
            s = s % 10;

            cur.next = new ListNode(s);
            cur = cur.next;
            if (n1 != null) {
                n1 = n1.next;
            }
            if (n2 != null) {
                n2 = n2.next;
            }
        }
        if (addV == 1) {
            cur.next = new ListNode(addV);
        }
        return head.next;
    }
}

class ListNode {
    int val;
    ListNode next;

    public ListNode(int val) {
        this.val = val;
    }

    public ListNode addFirst(int val) {
        ListNode node = new ListNode(val);
        this.next = node;
        return node;
    }

    public void display() {
        ListNode node = this;
        while (node != null) {
            System.out.println(node.val);
            node = node.next;
        }
        System.out.println();
    }
}
